//
//  AiChatBean.swift
//  imprint
//
//  Created by 小柠檬🍋 on 2025/2/27.
//

import Foundation
class AiConfigBean : Codable {
    var maxTokens: Int64? = nil
    var temperature: TimeInterval? = nil
    var top_p: TimeInterval? = nil
    var disableSystem: [String]? = nil//禁用结尾使用system，使用user
    
    //用于聊天结尾是用"system"还是user
    func getLastRoleType(mode: String?) -> String {
        // 如果 disableSystem 只有一个元素且为 "all"，返回 "user"
        if let disableSystem = disableSystem,
           disableSystem.count == 1,
           disableSystem.first == "all" {
            return "user"
        }
        
        // 如果 mode 为空，返回 "system"
        guard let mode = mode, !mode.isEmpty else {
            return "system"
        }
        
        // 检查 mode 是否在 disableSystem 中，存在则返回 "user"
        if let disableSystem = disableSystem, disableSystem.contains(mode) {
            return "user"
        }
        
        // 默认返回 "system"
        return "system"
    }
}
